Tidying up code.

Up to update price schedule