Fixed code styling

Signed-off-by: Aran Wilkinson <aran3001@gmail.com>