<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def add_ones(a):
 """Adds a column of 1s at the end of the array"""

    arr = N.ones((a.shape[0],a.shape[1]+1))
    arr[:,:-1] = a
    return arr