Changed the param so that it takes direct input